#include "compile_run.hpp"
#include <jsoncpp/json/json.h>
#include "../comm/httplib.h"
using namespace ns_complie_run;
using namespace httplib;

void Usage(char* s)
{
    std::cout << "Usage: " << s << " port" << std::endl;
}

int main(int argc, char* argv[])
{
    if(argc != 2)
    {
        Usage(argv[0]);
        exit(1);
    }

    Server svr;

    // svr.Get("/test", [](const Request& req, Response& resp){
    //     resp.set_content("这是一个测试", "text/plain;charset=utf8");
    // });
    svr.set_base_dir("./wwwroot");
    
    svr.Post("/compile_run", [](const Request& req, Response& resp){
        std::string in_json = req.body;
        std::string out_json;
        // std::cout << in_json << std::endl;
        if(!in_json.empty())
        {
            ComplieRun::Start(in_json, &out_json);
            // std::cout << out_json << std::endl;
            resp.set_content(out_json, "application/json;charset=utf8");
            // resp.body = out_json;
        }
    });

    svr.listen("0.0.0.0", std::atoi(argv[1]));








    // 以下为测试代码
    // Json::Value inval;
    // inval["code"] = R"(#include <iostream>
    //     int main()
    //     {
    //         std::cout << "hello, baichen!" << std::endl;
    //         // while(1);
    //         // while(1)
    //             //int* p = new int[1024 * 1024];
    //             // aaaaaaa
    //         // int* p = nullptr;
    //         // *p = 0;
    //         return 0;
    //     })";
    // inval["input"] = "";
    // inval["cpu_limit"] = 1;
    // inval["mem_limit"] = 30 * 1024;
    // Json::FastWriter writer;
    // std::string in_json = writer.write(inval);
    // std::string out_json;
    // std::cout << in_json << std::endl;
    // ComplieRun::Start(in_json, &out_json);
    // std::cout << out_json << std::endl;
    return 0;
}